package com.leetcode.algorithm.y22.m06;

/**
 * 309. 最佳买卖股票时机含冷冻期
 * https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/
 * 冷冻期为 1 天
 * 
 * @author jie.deng
 *
 */
class Question0309Solution01 {

	public int maxProfit(int[] prices) {
		// 数学建模：
		// 某一天买进股票，则当天的收益是 -prices[i]
		// 某一天卖出股票，则当天的收益是 +prices[i]

		// fail-fast
		int len = prices.length;
		if (len == 0) {
			return 0;
		}

		// DP状态定义:
		// dp[i][0] 表示第 i 天交易完，手上 没有 股票 && 非冷冻期 时的累计最大利润；
		// dp[i][1] 表示第 i 天交易完，手上 持有 股票时的累计最大利润。
		// dp[i][2] 表示第 i 天交易完，手上 没有 股票 && 冷冻期 时的累计最大利润。
		int dp[][] = new int[prices.length][3];
		// 初始化状态
		dp[0][0] = 0;
		dp[0][1] = -prices[0];
		dp[0][2] = Integer.MIN_VALUE; // 无效状态

		// 决策 a.买进 b.卖出 c.不操作
		// 状态转移方程
		for (int i = 1; i < prices.length; i++) {
			// dp[i][0]赋值
			// 前一天无股票，不操作
			dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]);

			// dp[i][1]赋值
			// 前一天有股票，不操作
			// 买进：前一天无股票 && 非冷冻期，今天买入
			dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);

			// dp[i][2]赋值
			// 卖出：前一天有股票，今天卖出
			dp[i][2] = dp[i - 1][1] + prices[i];
		}

		return Math.max(dp[prices.length - 1][0], dp[prices.length - 1][2]);
	}

}